import os
import random
import re
import time



# 获取该网页及后续分页下面的url地址
import requests
from selenium.webdriver import Firefox
from selenium.webdriver.common.keys import Keys  # 模仿键盘,操作下拉框的
from bs4 import BeautifulSoup  # 解析html的
from selenium.webdriver.firefox.options import Options

user_agent = [
    "Mozilla/5.0 (Macintosh; U; Intel Mac OS X 10_6_8; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50",
    "Mozilla/5.0 (Windows; U; Windows NT 6.1; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50",
    "Mozilla/5.0 (Windows NT 10.0; WOW64; rv:38.0) Gecko/20100101 Firefox/38.0",
    "Mozilla/5.0 (Windows NT 10.0; WOW64; Trident/7.0; .NET4.0C; .NET4.0E; .NET CLR 2.0.50727; .NET CLR 3.0.30729; .NET CLR 3.5.30729; InfoPath.3; rv:11.0) like Gecko",
    "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Trident/5.0)",
    "Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 6.0; Trident/4.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0)",
    "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1)",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.6; rv:2.0.1) Gecko/20100101 Firefox/4.0.1",
    "Mozilla/5.0 (Windows NT 6.1; rv:2.0.1) Gecko/20100101 Firefox/4.0.1",
    "Opera/9.80 (Macintosh; Intel Mac OS X 10.6.8; U; en) Presto/2.8.131 Version/11.11",
    "Opera/9.80 (Windows NT 6.1; U; en) Presto/2.8.131 Version/11.11",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_0) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Maxthon 2.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; TencentTraveler 4.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; The World)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; SE 2.X MetaSr 1.0; SE 2.X MetaSr 1.0; .NET CLR 2.0.50727; SE 2.X MetaSr 1.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; 360SE)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Avant Browser)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)",
    "Mozilla/5.0 (iPhone; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Mozilla/5.0 (iPod; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Mozilla/5.0 (iPad; U; CPU OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Mozilla/5.0 (Linux; U; Android 2.3.7; en-us; Nexus One Build/FRF91) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
    "MQQBrowser/26 Mozilla/5.0 (Linux; U; Android 2.3.7; zh-cn; MB200 Build/GRJ22; CyanogenMod-7) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
    "Opera/9.80 (Android 2.3.4; Linux; Opera Mobi/build-1107180945; U; en-GB) Presto/2.8.149 Version/11.10",
    "Mozilla/5.0 (Linux; U; Android 3.0; en-us; Xoom Build/HRI39) AppleWebKit/534.13 (KHTML, like Gecko) Version/4.0 Safari/534.13",
    "Mozilla/5.0 (BlackBerry; U; BlackBerry 9800; en) AppleWebKit/534.1+ (KHTML, like Gecko) Version/6.0.0.337 Mobile Safari/534.1+",
    "Mozilla/5.0 (hp-tablet; Linux; hpwOS/3.0.0; U; en-US) AppleWebKit/534.6 (KHTML, like Gecko) wOSBrowser/233.70 Safari/534.6 TouchPad/1.0",
    "Mozilla/5.0 (SymbianOS/9.4; Series60/5.0 NokiaN97-1/20.0.019; Profile/MIDP-2.1 Configuration/CLDC-1.1) AppleWebKit/525 (KHTML, like Gecko) BrowserNG/7.1.18124",
    "Mozilla/5.0 (compatible; MSIE 9.0; Windows Phone OS 7.5; Trident/5.0; IEMobile/9.0; HTC; Titan)",
    "UCWEB7.0.2.37/28/999",
    "NOKIA5700/ UCWEB7.0.2.37/28/999",
    "Openwave/ UCWEB7.0.2.37/28/999",
    "Mozilla/4.0 (compatible; MSIE 6.0; ) Opera/UCWEB7.0.2.37/28/999",
    # iPhone 6：
    "Mozilla/6.0 (iPhone; CPU iPhone OS 8_0 like Mac OS X) AppleWebKit/536.26 (KHTML, like Gecko) Version/8.0 Mobile/10A5376e Safari/8536.25",

]

hd = {'User-Agent': random.choice(user_agent)}


# 访问网页获取网页信息
def geturl(url):
    try:
        r = requests.get(url, timeout=30, headers=hd)
        r.raise_for_status()
        r.encoding = r.apparent_encoding
        return r.text
    except:
        return ''


# 通过目标网页及后续分页得到网页中视频连接写入txt(http://www.qimizb.com/mm/24/)
def getvideourl(url,pagno):
    count = 1
    # pagno = 60
    videourls = []
    try:
        while count <= pagno:
            videourltemp = []
            if count == 1:
                newurl = url
            else:
                newurl = url + str(count) + '.html'
            text = geturl(newurl)
            soup = BeautifulSoup(text, 'html.parser')
            div = soup.find('div', id="vlist")
            a = div.find_all("a")
            for tag in a:
                videourl = tag.attrs["href"]
                print(videourl)
                videourltemp.append(videourl)
            if len(videourltemp) == 0:
                print("last count: " + str(count))
                break
            else:
                videourls.extend(videourltemp)
            count += 1
    finally:
        dir_name = "E:/videobypy"
        if not os.path.exists(dir_name):
            os.mkdir(dir_name)
        for videourl in videourls:
            print(videourl)
            with open(dir_name + '/' + "videourl.txt", 'a+') as f:
                f.write(videourl + "\n")
        print(str(count))


# 通过目标网页及后续分页得到网页中视频连接写入txt(http://www.qimizb.com/tiaowu/1.html)
def getqimizbvideourl1(url, pagno):
    count = 1
    videourls = []
    try:
        while count <= pagno:
            videourltemp = []
            newurl = url[0:-6] + str(count) + '.html'
            text = geturl(newurl)
            soup = BeautifulSoup(text, 'html.parser')
            div = soup.find('div', id="newv")
            a = div.find_all("a", href=re.compile(".html"))
            print(newurl)
            for tag in a:
                videourl = tag.attrs["href"]
                videourltemp.append(videourl)
            if len(videourltemp) == 0:
                print("last count: " + str(count))
                break
            else:
                videourls.extend(videourltemp)
            count += 1
    finally:
        dir_name = "E:/videobypy"
        if not os.path.exists(dir_name):
            os.mkdir(dir_name)
        for videourl in videourls:
            print(videourl)
            with open(dir_name + '/' + "videourl.txt", 'a+') as f:
                f.write(videourl + "\n")
        print(str(count))


# 通过目标网页及后续分页得到网页中视频连接写入txt(http://www.qimizb.com/tiaowu/1.html)
def getweibovideourl(url,pagno):
    count = 1
    # pagno = 50
    videourls = []
    try:
        while count <= pagno:
            videourltemp = []
            newurl = url + str(count)
            options = Options()
            options.add_argument('-headless')
            driver = Firefox(executable_path='D:\\其他软件\\firfox\\geckodriver', firefox_options=options)
            print(newurl)
            driver.get(newurl)
            time.sleep(5)
            # driver.find_element_by_tag_name("video").send_keys(Keys.HOME)
            # time.sleep(1)

            driver.find_element_by_xpath("//video[@class='wbv-tech']").send_keys(Keys.DOWN)
            time.sleep(1)
            driver.find_element_by_xpath("//video[@class='wbv-tech']").send_keys(Keys.DOWN)
            time.sleep(1)
            # time.sleep(3)
            # print(driver.page_source)
            text = driver.page_source
            soup = BeautifulSoup(text, 'html.parser')
            video = soup.find_all('video', class_="wbv-tech")
            driver.close()
            for tag in video:
                videourl = tag.attrs["src"]
                print(videourl)
                videourltemp.append(videourl)
            # if len(videourltemp) == 0:
            #     print("last count: " + str(count))
            #     break
            else:
                videourls.extend(videourltemp)
            count += 1
    finally:
        driver.close()
        dir_name = "E:/videobypy"
        if not os.path.exists(dir_name):
            os.mkdir(dir_name)
        for videourl in videourls:
            print(videourl)
            with open(dir_name + '/' + "videourl.txt", 'a+') as f:
                f.write(videourl + "\n")
        print(str(count))


# 3月12号
if __name__ == "__main__":
    if os.path.exists("E:/videobypy/videourl.txt"):
        os.remove("E:/videobypy/videourl.txt")
    # getvideourl("http://www.qimizb.com/mm/56/")
    getqimizbvideourl1("http://www.qimizb.com/tiaowu/1.html", 6)
    # getweibovideourl(
    #     "https://s.weibo.com/video/%E5%BE%90%E9%9B%85&typeall=1&hasvideo=1&tw=video&page=")
